package com.imut.lagain.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.imut.lagain.entity.UserRole;
import com.imut.lagain.repository.UserRoleRepository;
import com.imut.lagain.service.IUserRoleService;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.time.LocalDateTime;
import java.util.List;

/**
 * 用户角色服务实现类
 */
@Service
@Transactional
@RequiredArgsConstructor
@Slf4j
public class UserRoleServiceImpl extends ServiceImpl<UserRoleRepository, UserRole> implements IUserRoleService {
    
    private final UserRoleRepository userRoleRepository;
    private final BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
    
    /**
     * 获取用户角色信息
     * @param userId 用户ID
     * @return 用户角色信息
     */
    @Override
    public UserRole getUserRole(Long userId) {
        log.info("开始获取用户角色信息: userId={}", userId);
        
        if (userId == null) {
            log.warn("用户ID为空");
            return null;
        }
        
        try {
            QueryWrapper<UserRole> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("user_id", userId);
            queryWrapper.eq("is_deleted", false);
            
            UserRole userRole = this.getOne(queryWrapper);
            log.info("查询用户角色结果: userRole={}", userRole);
            return userRole;
        } catch (Exception e) {
            log.error("查询用户角色信息异常: userId={}", userId, e);
            return null;
        }
    }
    
    /**
     * 根据角色类型获取用户角色列表
     * @param roleType 角色类型
     * @return 用户角色列表
     */
    @Override
    public List<UserRole> getRolesByType(String roleType) {
        if (!StringUtils.hasText(roleType)) {
            return null;
        }
        QueryWrapper<UserRole> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("role_type", roleType.toUpperCase());
        queryWrapper.eq("is_active", true);
        queryWrapper.eq("is_deleted", false);
        return this.list(queryWrapper);
    }
    
    /**
     * 创建用户角色
     * @param userId 用户ID
     * @param roleType 角色类型
     * @param roleName 角色名称
     * @return 用户角色信息
     */
    @Override
    public UserRole createUserRole(Long userId, String roleType, String roleName, String adminPassword) {
        if (userId == null || !StringUtils.hasText(roleType)) {
            log.warn("Invalid parameters for creating user role: userId={}, roleType={}", userId, roleType);
            throw new IllegalArgumentException("用户ID和角色类型不能为空");
        }
        
        // 检查是否已存在该用户的角色
        UserRole existingRole = getRoleByUserId(userId);
        if (existingRole != null) {
            log.warn("User role already exists: userId={}", userId);
            throw new RuntimeException("用户角色已存在");
        }
        
        UserRole userRole = new UserRole();
        userRole.setUserId(userId);
        userRole.setRoleType(roleType.toUpperCase());
        userRole.setRoleName(StringUtils.hasText(roleName) ? roleName : getRoleNameByType(roleType));
        userRole.setIsActive(true);
        userRole.setIsDeleted(false);
        
        boolean saved = this.save(userRole);
        if (saved) {
            return userRole;
        } else {
            log.error("Failed to create user role: userId={}, roleType={}", userId, roleType);
            throw new RuntimeException("创建用户角色失败");
        }
    }
    
    /**
     * 更新用户角色
     * @param userId 用户ID
     * @param roleType 角色类型
     * @param roleName 角色名称
     * @return 是否更新成功
     */
    @Override
    public boolean updateUserRole(Long userId, String roleType, String roleName) {
        if (userId == null) {
            return false;
        }
        
        UserRole userRole = getRoleByUserId(userId);
        if (userRole == null) {
            return false;
        }
        
        if (StringUtils.hasText(roleType)) {
            userRole.setRoleType(roleType.toUpperCase());
        }
        if (StringUtils.hasText(roleName)) {
            userRole.setRoleName(roleName);
        }
        userRole.setUpdateTime(LocalDateTime.now());
        
        return this.updateById(userRole);
    }
    
    /**
     * 验证管理员密码
     * @param userId 用户ID
     * @param password 密码
     * @return 是否验证成功
     */
    @Override
    public boolean validateAdminPassword(Long userId, String password) {
        if (userId == null || !StringUtils.hasText(password)) {
            return false;
        }
        
        UserRole userRole = getRoleByUserId(userId);
        if (userRole == null || !"ADMIN".equals(userRole.getRoleType()) || !userRole.getIsActive()) {
            return false;
        }
        
        // 管理员密码验证现在在User表中进行，这里直接返回false
        return false;
    }
    
    /**
     * 更新管理员密码
     * @param userId 用户ID
     * @param oldPassword 旧密码
     * @param newPassword 新密码
     * @return 是否更新成功
     */
    @Override
    public boolean updateAdminPassword(Long userId, String oldPassword, String newPassword) {
        if (userId == null || !StringUtils.hasText(oldPassword) || !StringUtils.hasText(newPassword)) {
            return false;
        }
        
        // 管理员密码现在存储在User表中，这里直接返回false
        return false;
    }
    
    /**
     * 获取所有用户角色列表
     * @return 用户角色列表
     */
    @Override
    public List<UserRole> getAllRoles() {
        QueryWrapper<UserRole> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("is_deleted", false);
        return this.list(queryWrapper);
    }
    
    /**
     * 根据用户ID获取角色
     * @param userId 用户ID
     * @return 用户角色
     */
    @Override
    public UserRole getRoleByUserId(Long userId) {
        if (userId == null) {
            return null;
        }
        QueryWrapper<UserRole> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("user_id", userId);
        queryWrapper.eq("is_deleted", false);
        return this.getOne(queryWrapper);
    }
    
    /**
     * 根据角色ID获取角色
     * @param roleId 角色ID
     * @return 用户角色
     */
    @Override
    public UserRole getRoleById(Long roleId) {
        if (roleId == null) {
            return null;
        }
        return this.getById(roleId);
    }
    
    /**
     * 更新用户角色
     * @param roleId 角色ID
     * @param roleType 角色类型
     * @param roleName 角色名称
     * @param isActive 是否激活
     * @return 是否更新成功
     */
    @Override
    public boolean updateRole(Long roleId, String roleType, String roleName, Boolean isActive) {
        if (roleId == null) {
            return false;
        }
        
        UserRole role = this.getById(roleId);
        if (role == null) {
            return false;
        }
        
        if (StringUtils.hasText(roleType)) {
            role.setRoleType(roleType.toUpperCase());
        }
        if (StringUtils.hasText(roleName)) {
            role.setRoleName(roleName);
        }
        if (isActive != null) {
            role.setIsActive(isActive);
        }
        role.setUpdateTime(LocalDateTime.now());
        
        return this.updateById(role);
    }
    
    /**
     * 删除用户角色（逻辑删除）
     * @param roleId 角色ID
     * @return 是否删除成功
     */
    @Override
    public boolean deleteRole(Long roleId) {
        if (roleId == null) {
            return false;
        }
        
        UserRole role = this.getById(roleId);
        if (role == null) {
            return false;
        }
        
        role.setIsDeleted(true);
        role.setUpdateTime(LocalDateTime.now());
        return this.updateById(role);
    }
    
    /**
     * 根据角色类型获取默认角色名称
     * @param roleType 角色类型
     * @return 默认角色名称
     */
    private String getRoleNameByType(String roleType) {
        switch (roleType.toUpperCase()) {
            case "ADMIN":
                return "管理员";
            case "PARTNER":
                return "伴侣";
            default:
                return "普通用户";
        }
    }
    
    /**
     * 设置用户角色
     * @param userId 用户ID
     * @param roleType 角色类型
     * @param roleName 角色名称
     * @return 用户角色
     */
    @Override
    public UserRole setupUserRole(Long userId, String roleType, String roleName, String adminPassword) {
        // 检查是否已存在角色
        UserRole existingRole = getRoleByUserId(userId);
        if (existingRole != null) {
            // 更新现有角色
            existingRole.setRoleType(roleType.toUpperCase());
            existingRole.setRoleName(StringUtils.hasText(roleName) ? roleName : getRoleNameByType(roleType));
            existingRole.setUpdateTime(LocalDateTime.now());
            
            boolean updated = this.updateById(existingRole);
            return updated ? existingRole : null;
        } else {
            // 创建新角色
            return createUserRole(userId, roleType, roleName, adminPassword);
        }
    }
    
    @Override
    public boolean setAdminPassword(Long userId, String password) {
        return false;
    }
    
    @Override
    public List<UserRole> getAllAdmins() {
        return getRolesByType("ADMIN");
    }
    
    @Override
    public List<UserRole> getAllPartners() {
        return getRolesByType("PARTNER");
    }
    
    @Override
    public boolean updateRoleStatus(Long userId, Boolean isActive) {
        UserRole userRole = getRoleByUserId(userId);
        if (userRole == null) {
            return false;
        }
        userRole.setIsActive(isActive);
        userRole.setUpdateTime(LocalDateTime.now());
        return this.updateById(userRole);
    }
    
    @Override
    public boolean deleteUserRole(Long userId) {
        UserRole userRole = getRoleByUserId(userId);
        if (userRole == null) {
            return false;
        }
        userRole.setIsDeleted(true);
        userRole.setUpdateTime(LocalDateTime.now());
        return this.updateById(userRole);
    }
    
    @Override
    public boolean hasRole(Long userId, String roleType) {
        UserRole userRole = getRoleByUserId(userId);
        return userRole != null && roleType.equalsIgnoreCase(userRole.getRoleType());
    }
    
    @Override
    public String getUserRoleType(Long userId) {
        UserRole userRole = getRoleByUserId(userId);
        return userRole != null ? userRole.getRoleType() : null;
    }
    
    @Override
    public boolean initializeAdminRole(Long userId, String password) {
        return false;
    }
    
    @Override
    public boolean isAdmin(Long userId) {
        return hasRole(userId, "ADMIN");
    }
    
    @Override
    public boolean isPartner(Long userId) {
        return hasRole(userId, "PARTNER");
    }
}